(Python) Can't get matrix scalar multiplication output correct